import numpy as np

# 生成一个示例数组
arr = np.arange(1, 21).reshape(4, 5)
print("原始数组:")
print(arr)

# 使用 np.save 存储数组为 .npy 文件
np.save('45.npy', arr)
print("\n数组已存储为 45.npy")

# 使用 np.load 加载 .npy 文件中的数组
loaded_arr = np.load('45.npy')
print("\n从 45.npy 加载的数组:")
print(loaded_arr)

# 生成第二个示例数组
arr2 = np.random.rand(3, 3)

# 使用 np.savez 存储多个数组为 .npz 文件
np.savez('multiple_arrays.npz', arr1=arr, arr2=arr2)
print("\n多个数组已存储为 multiple_arrays.npz")

# 使用 np.load 加载 .npz 文件中的数组
loaded_data = np.load('multiple_arrays.npz')
print("\n从 multiple_arrays.npz 加载的数组:")
print("arr1:")
print(loaded_data['arr1'])
print("arr2:")
print(loaded_data['arr2'])

# 使用 np.savetxt 存储数组为文本文件
np.savetxt('output_array_text.txt', arr, fmt='%d', delimiter=',')
print("\n数组已存储为 output_array_text.txt")

# 使用 np.loadtxt 加载文本文件中的数组
loaded_txt_arr = np.loadtxt('output_output_array_text.txt', dtype=int, delimiter=',')
print("\n从 output_output_array_text.txt 加载的数组:")
print(loaded_txt_arr)
